import site
import os
import pkg_resources

# 获取 site-packages 目录
site_packages_dirs = site.getsitepackages()

print("site-packages目录:", site_packages_dirs)

# 检查 Flask 的安装位置
package_name = 'flask'

# 遍历 site-packages 目录，目的是将列表中的数据取出，列表中只有一个元素也可以使用列表索引的方式，然后判断目录下是否存在 Flask
for dir in site_packages_dirs:
    package_path = os.path.join(dir, package_name)

    if os.path.exists(package_path):
        print(f"{package_name} installed at: {package_path}")
    else:
        # 也可以使用 pkg_resources 来获取包的路径
        dist = pkg_resources.get_distribution(package_name)
        print(f"{package_name} installed at: {dist.location}")
